from PIL import Image
# Pillow

"""
照片填充并切割成九宫格形式，用于发微信朋友圈
PIL是python2的包
python3之后使用的是Pillow pip3 install Pillow
但是代码里还是要用PIL声明
"""

# 先将 input image 填充为正方形
def fill_image(image):
    width, height = image.size
    # 选取长和宽中较大值作为新图片的边长
    new_image_length = width if width > height else height
    # 生成新图片[白底]，底色可配置其他颜色
    new_image = Image.new(image.mode, (new_image_length, new_image_length), color='white')
    # 将之前的图片input image 粘贴在新图上，居中
    if width > height:# 原图宽大于高，则填充图片的竖直维度  #(x,y)二元组表示粘贴上图相对下图的起始位置,是个坐标点。
        new_image.paste(image, (0, int((new_image_length - height) /2)))
    else:
        new_image.paste(image, (int((new_image_length - width) /2), 0))
    return new_image

def cut_image(image):
    width, height = image.size
    item_width =int(width /3)# 因为朋友圈一行放3张图
    box_list = []
    # (left, upper, right, lower)
    for i in range(0, 3):
        for j in range(0, 3):
            box = (j*item_width, i*item_width, (j+1)*item_width, (i+1)*item_width)
            box_list.append(box)

    image_list = [image.crop(box) for box in box_list]
    return image_list

# 保存
def save_images(image_list):
    index =1
    for image in image_list:
        image.save(str(index) +'.png', 'PNG')
        index +=1

if __name__ =='__main__':
    file_path ="/Users/mac/Downloads/P90309-231054.jpg"    # 把目标图片 input image 放到代码所处的文件夹里
    image = Image.open(file_path)
    image = fill_image(image)
    image_list = cut_image(image)
    save_images(image_list)
